from src.config.default import _CN as cfg

cfg.LGFCTR.ENCODER.LAYER_NAMES = [['self']*2, ['self']*4, ['self']*6]

cfg.LGFCTR.DECODER.LAYER_NAMES = [['cross']*6, ['cross']*4, ['cross']*2]
cfg.LGFCTR.DECODER.LAYER_TYPES = ['multi'] * 2 + ['CNN']

cfg.LGFCTR.MATCH_COARSE.MATCH_TYPE = 'dual_softmax'
cfg.LGFCTR.MATCH_COARSE.SPARSE_SPVS = True
cfg.LGFCTR.MATCH_COARSE.TRAIN_COARSE_PERCENT = 0.3

cfg.TRAINER.WARMUP_STEP = 368 * 100 * 3 // 6 // (64//6)  # 3 epochs
